#!/bin/bash

# 设置颜色变量
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
NC='\033[0m' # No Color

# 检查是否已安装 Python
if ! command -v python3 &> /dev/null; then
    echo -e "${RED}错误: 未找到 Python3，请先安装 Python3${NC}"
    exit 1
fi

# 检查 Python 版本
python_version=$(python3 --version)
echo -e "${GREEN}当前 Python 版本: $python_version${NC}"

# 检查虚拟环境是否已存在
if [ -d "venv" ]; then
    echo -e "${YELLOW}发现已存在的虚拟环境，正在删除...${NC}"
    rm -rf venv
fi

# 创建新的虚拟环境
echo -e "${GREEN}正在创建新的虚拟环境...${NC}"
python3 -m venv venv

# 激活虚拟环境
echo -e "${GREEN}正在激活虚拟环境...${NC}"
source venv/bin/activate

# 升级 pip
echo -e "${GREEN}正在升级 pip...${NC}"
python -m pip install --upgrade pip

# 安装依赖
echo -e "${GREEN}正在安装项目依赖...${NC}"
pip install -r requirements.txt

echo -e "${GREEN}环境设置完成！${NC}"
echo -e "${GREEN}虚拟环境已激活，可以开始工作了！${NC}"
echo -e "${YELLOW}提示：退出虚拟环境请输入 'deactivate'${NC}"